# from huggingface_hub import snapshot_download

# # 指定你希望的本地路径
# local_model_path = "./models/rtdetr_r50vd_coco_o365"

# # 下载模型
# snapshot_download(
#     repo_id="PekingU/rtdetr_r50vd_coco_o365",
#     local_dir=local_model_path,
#     local_dir_use_symlinks=False  # 不使用软链接，直接复制文件
# )


# 2. 加载并保存模型
from ultralytics import YOLO
import shutil, os

# 2.1 指定你想存放模型的目录（不存在会自动创建）
save_dir = "./models/yolo"          # 改成你的路径
os.makedirs(save_dir, exist_ok=True)

# 2.2 第一次调用 YOLO("yolov8n.pt") 时，ultralytics 会自动下载
model = YOLO("yolov8n.pt")               # 加载 nano 版本

# 2.3 把下载好的权重文件拷贝/移动到目标目录
src = "yolov8n.pt"                       # 下载后默认放在当前工作目录
dst = os.path.join(save_dir, "yolov8n.pt")
shutil.move(src, dst)                    # 如果想保留两份，用 shutil.copy

print("YOLOv8-nano 已保存至：", dst)